Fetching 6 files: 100%|██████████████████████████████████████████████████████████████████████████| 6/6 [00:00<?, ?it/s]
flux-2-klein-base-9b-Q6_K.gguf: 100%|█████████████████████████████████████████████| 7.87G/7.87G [03:07<00:00, 41.9MB/s]
Traceback (most recent call last):
  File "C:\Users\Admin\Documents\LLM\OneTrainer_f\modules\modelLoader\Flux2ModelLoader.py", line 180, in load
    self.__load_internal(
  File "C:\Users\Admin\Documents\LLM\OneTrainer_f\modules\modelLoader\Flux2ModelLoader.py", line 53, in __load_internal
    raise Exception("not an internal model")
Exception: not an internal model

Traceback (most recent call last):
  File "C:\Users\Admin\Documents\LLM\OneTrainer_f\modules\modelLoader\Flux2ModelLoader.py", line 188, in load
    self.__load_diffusers(
  File "C:\Users\Admin\Documents\LLM\OneTrainer_f\modules\modelLoader\Flux2ModelLoader.py", line 79, in __load_diffusers
    transformer = Flux2Transformer2DModel.from_single_file(
                  ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "C:\Users\Admin\Documents\LLM\OneTrainer_f\venv\Lib\site-packages\huggingface_hub\utils\_validators.py", line 114, in _inner_fn
    return fn(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^
  File "C:\Users\Admin\Documents\LLM\OneTrainer_f\venv\src\diffusers\src\diffusers\loaders\single_file_model.py", line 491, in from_single_file
    load_model_dict_into_meta(
  File "C:\Users\Admin\Documents\LLM\OneTrainer_f\venv\src\diffusers\src\diffusers\models\model_loading_utils.py", line 291, in load_model_dict_into_meta
    hf_quantizer.check_quantized_param_shape(param_name, empty_state_dict[param_name], param)
  File "C:\Users\Admin\Documents\LLM\OneTrainer_f\venv\src\diffusers\src\diffusers\quantizers\gguf\gguf_quantizer.py", line 84, in check_quantized_param_shape
    raise ValueError(
ValueError: double_stream_modulation_img.linear.weight has an expected quantized shape of: (24576, 4096), but received shape: torch.Size([24576, 8192])

Traceback (most recent call last):
  File "C:\Users\Admin\Documents\LLM\OneTrainer_f\modules\modelLoader\Flux2ModelLoader.py", line 196, in load
    self.__load_safetensors(
  File "C:\Users\Admin\Documents\LLM\OneTrainer_f\modules\modelLoader\Flux2ModelLoader.py", line 167, in __load_safetensors
    raise NotImplementedError("Loading of single file Flux2 models not supported. Use the diffusers model instead. Optionally, transformer-only safetensor files can be loaded by overriding the transformer.")
NotImplementedError: Loading of single file Flux2 models not supported. Use the diffusers model instead. Optionally, transformer-only safetensor files can be loaded by overriding the transformer.

Traceback (most recent call last):
  File "C:\Users\Admin\Documents\LLM\OneTrainer_f\modules\ui\TrainUI.py", line 750, in __training_thread_function
    trainer.start()
  File "C:\Users\Admin\Documents\LLM\OneTrainer_f\modules\trainer\GenericTrainer.py", line 129, in start
    self.model = self.model_loader.load(
                 ^^^^^^^^^^^^^^^^^^^^^^^
  File "C:\Users\Admin\Documents\LLM\OneTrainer_f\modules\modelLoader\GenericLoRAModelLoader.py", line 51, in load
    base_model_loader.load(model, model_type, model_names, weight_dtypes, quantization)
  File "C:\Users\Admin\Documents\LLM\OneTrainer_f\modules\modelLoader\Flux2ModelLoader.py", line 205, in load
    raise Exception("could not load model: " + model_names.base_model)
Exception: could not load model: black-forest-labs/FLUX.2-klein-base-9B